import torch
torch.manual_seed(666)

data = torch.randint(1,10,size=(3,4,5))
print(data)

print('=' * 200)

# 获取0周上的第一个数据
print(data[0,:,:])

print('=' * 200)

# 获取1轴上的第一个数据
print(data[:,0,:])
print('=' * 200)
# 获取2轴上的第一个数据
print(data[:,:,0])
print(data[0])
